package com.my.lucky.repository;

import com.my.lucky.entity.StockHotRank;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Optional;

/**
 * 股票热度排名数据访问层
 */
@Repository
public interface StockHotRankRepository extends JpaRepository<StockHotRank, Long> {

    /**
     * 根据股票代码查询最新的热度排名
     *
     * @param stockCode 股票代码
     * @return 热度排名信息
     */
    Optional<StockHotRank> findFirstByStockCodeOrderByRankDateDesc(String stockCode);

    /**
     * 根据日期范围查询热度排名列表
     *
     * @param startDate 开始日期
     * @param endDate   结束日期
     * @return 热度排名列表
     */
    List<StockHotRank> findByRankDateBetweenOrderByRankDateDesc(LocalDateTime startDate, LocalDateTime endDate);

    /**
     * 根据关注度阈值查询热门股票
     *
     * @param attentionDegree 关注度阈值
     * @return 热门股票列表
     */
    List<StockHotRank> findByAttentionDegreeGreaterThanOrderByAttentionDegreeDesc(Integer attentionDegree);


    /**
     * 查询指定日期范围内的最大榜单时间
     *
     * @param startDate 开始日期
     * @param endDate   结束日期
     * @return 最大创建时间
     */
    @Query("SELECT MAX(s.rankDate) FROM StockHotRank s WHERE s.rankDate >= :startDate AND s.rankDate < :endDate")
    Optional<LocalDateTime> findMaxCreateTimeBetween(@Param("startDate") LocalDateTime startDate,
                                                     @Param("endDate") LocalDateTime endDate);

    /**
     * 根据指定榜单时间查询数据
     *
     * @param rankDate 榜单时间
     * @return 指定榜单时间的数据列表
     */
    List<StockHotRank> findByRankDate(LocalDateTime rankDate);

    /**
     * 根据指定榜单时间查询数据（按关注度降序排列）
     *
     * @param rankDate 榜单时间
     * @return 指定榜单时间的数据列表
     */
    List<StockHotRank> findByRankDateOrderByAttentionDegreeDesc(LocalDateTime rankDate);

    /**
     * 根据指定榜单时间和股票代码查询数据
     *
     * @param rankDate  榜单时间
     * @param stockCode 股票代码
     * @return 指定榜单时间和股票代码的数据
     */
    Optional<StockHotRank> findByRankDateAndStockCode(LocalDateTime rankDate, String stockCode);

    /**
     * 根据指定榜单时间范围查询数据（按榜单时间和关注度排序）
     *
     * @param startDate 开始榜单时间
     * @param endDate   结束榜单时间
     * @return 指定时间范围的数据列表
     */
    List<StockHotRank> findByRankDateBetweenOrderByRankDateDescAttentionDegreeDesc(LocalDateTime startDate, LocalDateTime endDate);

}